from django.db import models

from django.contrib.auth.models import AbstractUser
# Create your models here.


class User(AbstractUser):
     # 额外增加 mobile 字段
    mobile = models.CharField(max_length=11,
                              unique=True,
                              null=True,
                              verbose_name='手机号')
    invitation = models.CharField(max_length=8,
                                  null=True,
                                verbose_name='邀请码')
    myinvitation = models.CharField(max_length=8,
                                    unique=True,
                                    null=True,
                                verbose_name='我的邀请码')
# 对当前表进行相关设置:
    class Meta:
        db_table = 'tb_users'
        verbose_name = '用户'
        verbose_name_plural = verbose_name

    # 在 str 魔法方法中, 返回用户名称
    def __str__(self):
        return self.username